Add back in local storage functionality