Fixed minor issues, typos and comments.
Added some tests for the JugglingModelTrait and fixed some errors.