Updated script to include more sample data