Made some UI fixes