Refactored draft edit page