Finished part 2
Added test input