FSM visualisation script