Updated if statemens