Changed the code sample