completed the new class diagram