Finishing the new visualiztion, and improving its UI