Working code. Performing better than Neo4j