I am trying to re-use the code for implementing a a regular DQN. So could you guide me what updates needs to be done for the same?? Thanks