How to convert string to list using the split() function in Python