Added related work section. Added comments