updates to edit page and overall UI