First version of visualization module