fix up local storage and UI