Stored the map into the pregel local global collection. Added tests for the callback after execution of all tasks has finished to send messages and active vertices. They fail though