Adding afterFind and updateDb