Reloading a model now works, fix for issue #2