Updated the code samples