Solution to warmup