Improvements to edit post