Tweak styles and make main button work