add button and more misc styling