Update list_compr_2.py