Also fixed the Dockerfile