Add refresh logic