Styling changes
Add remove button