Modified run.py for AWS'