Make the for loops python'ish