Ashkan: Visualization code is added