oops didn't comment out visualization code