Working with UML to organize