Added code to pass should return 6 when i = 3