Code style: replace procedure syntax with Unit return type

See http://docs.scala-lang.org/style/declarations.html#procedure_syntax