Working on saving the data in DB