Improved button slightly