Cleaned up the Dockerfile