Add class info function