Last min tweaks