Edited Related Work section