little code polishing