Add generation test