Changed the format of the code to follow closer with what Wolberg posted on the class website.