Working with the final visualization sketch

Made it object oriented and stuff.