Add code to edit posts