Is there a simpler way to write this if