EDIT ui updated with new buttons