removed "bug" with reward. Attention, currently it is cheaper for an agent to wait if we cummulate rewards between the different state!