continued the assignment and added some documentation