works on initial answer for assignment 1