Fixed mistakes in models