Clean up sudoku.py