added some UI features