few improvements to the code balance experiments