Add set button functionality