Adding FSM diagram