Adding script that visualizes the results of problem 2