Updated script to work with actual data