Moved all visualization stuff to visualization/