add class
Revert "add class"

This reverts commit e37aa7c04ce90f0b8f9b4a0fee0f50dd2b25540d.