preparing for pypi : simplifying structure of module
 small fix in the experiment