For sure! Yea, it's tricky since there's rotation, maybe when ball velocity gets very small you could set it to zero directly in the code, similar for angular velocity, but /shrug as well :)
