Added related work section