Add styles and spruce up a little