It's a lot more readable like this if you have multiple returns imo.