Add submitted code for part 1