Add first attempt at assignment1