added new class diagram and fixed failing test in roundtest