Added logging of episodic reward and total reward.