Code tweaks
Code tweaks