Combine if statement with 3 different return