Change generate_data.py script to include ratings