Remove `i = 1` and replace `while i < WIDTH` with `WIDTH.times do |i|`