And updated the run.py with some documentation aswell.