Moved cassify classes from test to main