Update related work section