visualization working + more stuff