add support for different types of graphs
fixing test, sorry world :(
Merge pull request #17 from vguardiola/master

add support for different types of graphs