Extract reward functionality to QState