Add FSM diagram