Significantly clean up class transformer.

Signed-off-by: Mitch Petrie <mitchpetrie29@gmail.com>