code tweaks
Tweaks