Added first try of class diagram