Fixed comments and added missing parameters to model methods