Add edit mode for post and comment