Code Re-Usability

https://github.com/harshraj22/rl_lab/blob/b51fb302058d31c30282160db1cc29cc3ad65884/lab2/utils/utils.py#L51

New classes are being needed to be created for new agents.
Either the agents needs are completely different, or the base class requirements hasn't been identified properly. 