Added more functionality to RLLibViz