Added script which plots loss