Making a conveyor belt system in python