Added UML for state pattern