Improved model architecture.