Small changes to if then else.