made visualization code it's own file. axis labels better