Adding back cooldown