Add an option to export models