Add learn button support