Use better if statement