Define A Custom Code In Python