Переменная Tkinter Checkbutton не изменяется на втором уровне окна

Question

У меня есть кнопка в окне Tkinter, которая ведет меня ко второму окну Tkinter с некоторыми кнопками Checking. Если я нажимаю на них, их назначенная переменная не меняется. Второе окно само по себе работает нормально.

test2.py

from Tkinter import *
import test

window = Tk()
btn = Button(window, command=test.main)
btn.pack()
window.mainloop()

test.py

from Tkinter import *
from functools import partial

def func(n):
  print var[n].get()


def main():
  window = Tk()
  global var 
  var = [IntVar(), IntVar()]

  i = 0
  ck = Checkbutton(window, variable=var[i], command=partial(func,i))
  ck.grid(row = 0, column = 0)

  i += 1
  ck2 = Checkbutton(window, variable=var[i], command=partial(func,i))
  ck2.grid(row = 1, column = 0)

  window.mainloop()

# main()