Use implicit return where possible