added class names to match css