Add support to save all models