Update comments

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>
Making sure behavior is consistent wrt Py2/Py3

Range returns an iterable in Python3. Making sure result is a list
always.

Signed-off-by: Jonas Kalderstam <jonas@kalderstam.se>