I got data visualization working kind of