Added custom class for demo