Added refresh button. Small refactor.