added class diagram (i don't know what's inside)