Individual rewards

Unified the type signatures of the reward and observation functions.