Python Comparing Numbers For K