HW02 - Got exercise 4 working!