How to modify the code to read dataset? I'm new to this, so I'm not very proficient in some aspects, thank you.