Added copy of code to modify for first exercise