Hrm, still can't figure out how to separate classes. Might need better data.