Bugfix + tweaks