started updating FSM for additional functionality