Updated box method in grid class