Update XML after classes have been generated