added modification to ui