Added different dimensions for real dataset
Finding the state names all by itself now