Working part 1
Finished 2