caffeination function added in