added some UI