Add class annotations tests