Move import code to build.sh