Creating martingale.py