Adding UI fixes