Added refresh on button click