data structures and algorithm in python