Add refresh() behavior