Rewrite xml using python ET