Set main loop in mastermind.py