For loop not iterating in the function- Language: Python