small changes to UI