UI fixes (#64)