Add some tests

Improve display_drb