Adding ability to save models