Python: Concocting two lists with function and for loop.