Fix up new test for save()/load(). Addresses issue #18.
Fix up save() by saving class of model. Addresses issue #18.