Adding first attempt at assignment 3 - output is not correct