Add in the code supporting the refresh button