Fixing bug where prep_data needs to be called twice. Allowing for different hidden layer sizes

fixed imports

updated training and inference script

added root_gnn_dgl directory