added image of class diagram