Add initial plotting poc