Add flag to export model into file