Added scenario.py which I stupidly forgot to include :P