Add db write function