Implement ransac in python