You should also update `display()` method.