Add work so far on prblem 1