Updated class names to match changes in demo.php