Added top comments, updated UML