Add models for new functionality.