Add a reload function