Added some corrections in models