Improved the docker build file