Some tweaks
If/else optimisation