Add edit post function