Update code to prod