# Clean and optimized ball tracking logic