Possible solution to 10
Get solution to problem 10 working