Updated reload function