First version of visualization part