Bugfix + improuvements