Branched off to work on finishing problem 2
Finished 2b