data structures and some problems in python