Added py file to Dockerfile