python input list into function