Code tweaks.