Cleanup
Remove unused variable
Train the last layer unless impossible