Updated Models section and related work

As the title says