Added the return model to the puts.
Merge pull request #234 from Jedwondle/master

Added the return model to the puts.