Styling needs work, but the done button should be right column