Add a bit more scala to the example