Update comments in QLearning class