stupid pick_action. added p=list(policy) in np.random.choice