Add class retrieval