Change this to pythonic way please (not looping over i)