revamping the UI