fixed some UI parts