Added work on pretrained models