Added but didn't implement load and save feature, added comments, changed UML diagram to reflect changes