improve statemachine visualisation