fixed UI problems

Fixed some UI issues