Revert "removed unused class"

This reverts commit 093fa6f0a53bd031a09e4042c3aa25860bc947e5.