Fixed some Errors with the models