1. Add delegate for the class.
2. Fix a display bug.