Write A Simple Python Code