def remainder in python - Search