Combine into one if statement?