Adding team model tests