improve the little feature