Python Caffeine Calculator