Very rough draft of State diagram