Flushed out unit tests for the model