minor changes

add reward component
change the path
few plots