Modified graph to have start and cur state. discover-reward makes initial policy pi from state and action space