modify button set method