Made things better

Cooldown works now