computing polynomial equation in python