Add ability to toggle between showing and editing states