added more functions to classes in uml diagram