Modified for Assignment 2

Made some changes to get needed results for Assignment 2.