Add more transitions and ability to graph the states and transitions