Same exercise with one line of code