I don't think this turned out so good. I probably need to remake the dataset. But for now, I'll publish it for testing.