Yes!. It is working now. Only major changes you did are that you didn’t do the train_test_split and used expand() instead of  view(). Is there anything else that could have caused this issue?
Thanks for your help