Updates to Dockerfile and run.py