Can add a choice input to select what state follows certain steps. This would require storing every state function created.