removed unused code
and edited UML