Added database integration