UI + bugfixes