Adding documentation to homework assignment 1