Removed manual testing code
Added tests for graph