Sligth tweaks