Python Brute Force Algorithm