added UML and removes rest