Added warmup

Added warmup code to help with the class work