Update the code reference to cover new features
Update the code reference to cover new features