Added a Database class for storage