Added functionality to reset buttons