Styling
Add classes to cancel and done buttons, call done()
Styling