little code improve