added tests around remember code functionality