Why is this python loop not working as I expect