updated FSM and TB for hw2