Better scala styling in exercise 3