Fixed some UI bugs