fixing up the ui