first python pgm :)