added database class (forget to add it before)