Update caffeine code