Better version of contextual bandit