Adding change return [Python]