Tweak Dockerfile