small project on deep reinforcement Learning

I have implemented Relational Deep Reinforcement Learning framework tested on boxworld (See DeepMind "Relational Deep Reinforcement Learning" paper (2018)). The paper uses an actor-critic method but this is a single player. What I want to do now is add one more agent and that the reward can be shared and see if the agent would learn faster.