Set button logic complete