Example For Python Function For Square