U can change small RL code to do this