modify a little code