Updates to related work and algorithm description