Completed HW1. Need to make it clean and efficient