Add thing for state compactness
Add scatterplots