adding q-learning