Added related work and citations