cleaning up for class