a class diagram added!