class diagram fix hopefully