some code for python learning