Add solution code section and minor updates

The solution code is under scala/solution/. I added a section at the bottom to with a pointer to that folder.

Also included some updates based on Patrick's comments.