python input list as argument