Some parameters and visualization code added.