code using Python