Adding the code to edit a post