small fix on related work, still needs a complete overhaul