cleaning up

for drawing class diagram