Add unit tests for new model fields