Use execute_concurrent_with_args for loadbalancing tests
PEP-8 fixes