first draft of list_comp.py