Add cooldown argument