fixed subgraph creation
added small test dataset
added some tests for small dataset