Bugfixes und cleanup