Create N-bandit.py