Add rerender method