Added data visualization (STILL WORK IN PROGRESS)