fix button in add class