few tweaks

few tweaks