Added code from class 03 Sep