Add reset function & refresh button