Improving source code as I learn more about Scala.