Improve button module