Add more conditions to 'if' statement