Add record_reader to setup