Adding tests for the base model