Added UML for story