Fix issue where memory update at end may raise an exception when processing non-bipartite graphs

[daniel-gomm]

This pull request fixes issue #29.

Instead of updating the memory separately for sources and destinations, both sets of nodes are updated at once, similar to the update_at_start strategy