Create zebra_puzzle.py