add a little new code