Add equality tests for models