Style tweaks, added a button