bug & UI fixes