Bugfixes & slick UI