Implementing the Q_Learner's get_action method