Added draft for class diagram