Separate the data and visualization class