Reviews: State Aware Imitation Learning