add visualization code and minor changes to the experiments