update code / test