Adding a cooldown feature