A little progress on FSM assignment