edit ui for more flexible usability