Update playlist_generator.py