Adding the basic working Python Code