Trying to add edit mode on click