passing list as argument in python to a function