changed everything. reduced to 3 classes