restore functionality to buttons