[mb] Updating maximum balls python code