Using xrange from six

In python 2 vs 3 xrange is different. This fix is an enhancement
to use xrange from six, instead of additional logic of handling
xrange in python 2 vs python 3.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>