Add ability to toggle between edit views