Some minor changes in code. Creating a visualiser for proper displaying of results.
TODO: Test the contour feature.