Geeks For Geeks Function In Python Practice