Adding changes to the if helper