make better button
fix the button