adding some features in the ui