Change name of function to create visualization