Wrap the code in a function so we get implicit returns