Currently the exercise is using Java style when dealing with null argument. Change it to use Scala way of dealing with null arguments.