Bugfixes / tweaks