Minor changes to header and development of database class