Create StateAndReward.java