add pickling test
fix units
added pickle test
pickle model then return