Use should matcher instead of assert
Add SimpleGraphItem trait for classes with simple key
Work on tests