Updated code to match Getting Started tutorial.