move some code for button