Small changes to "if" statement