Balls collision works but not very correctly