update code

update for the OSR project