Added a bit more to related work, Problem statement section added