added model and tests pass