udpate code for build