Added code to parse solution and visualize