Improve documentation for sudoku.py