Add demo. Clean up