Added else statement (had forgotten previously)