Make if statement one-liner