Add plumbing for use with Q-Learning