Rewrite the Dockerfile to simplify the setup