modified to follow updated tutorial
modified to follow updated tutorial