Completed part 1 during class
Don't know why I have this...
Added notebook theme
Solved second part naively, but it makes sense to me...