Take List As An Input In Python