added functions to prep and vizulaize data