Python Code Linear Search