Bugfixes & updates on UI