Added entities for connection to database