Multiple For Loop In Python Using Return