Better styling
Add a toogle button to change between edit and preview