added end state to controller. possibly final push for assignment 2.