Added the visualization engine. Need to add matplotlib and all and generate pretty visualizations