fix small bugs with UI