Bugfixes and lose screen