Better conditional statement