improved model registration