Add model tests for new methods