Fix: Update script and Dockerfile