Change logic in method "generator"