Added warm up code