adding some UI features