Merge record() and record_and_get_data()

Find a way to merge both function called before and after call, respectively.