[BUG] Fix model export
[FEATURE] Add controller