Updates to model save function...

Closes #136...