the counting case is done (ish) have a lot of tweaks, want to find classes and add correlation, test