added the start of a uml diagram for part 2