Add in model tests