improve classification