Add working code with sample data