Improved the dockerfile