Fixed code example

Fixed the code example.