branch code addition

just made this update for class