Add more functionality to the Database class