Clean up classification step integration