Update StateAndReward.java