Change error handle method
Button class done!