Thompson Sampling - Thompson Sampling in Python - Step 2