Add two missing methods to the `Database` class