fix + tweaks