Updated 2 files to based on professor advice

Changed 2 files