Messing around with Sudoku in Python