writing program in python