Attempting to get generation to match examples. Added failing test