Updated code to match Getting Started Tutorial.