update code for build