added support to save save model