HW 1 Code Version 1

First attempt at code for HW 1