First attempt at Multi-Class solver