Updated demo code and styling.