specify range in for loop python