code in book "data structure and algorithm using python"