1st attempt at assignment 2