a little code update