Added all new code for the vizualizer