Class tidy up