Add generate shifts to GenerateDataCommand