Add return to draft button