added capsule classifier