Added refresh functionality. Hopefully.