fixing button classes