Adding warmup functionality