Add exploratory tests to get models working