Implement early version of visualization