Small changes to make python script work more often.
Merge pull request #421 from alexjacks92/pyScript

Small changes to make python script work more often.