Adding puzzle model tests