added main in python program