Made visualization white background, cleaned up training files.