Added more code to Assignment 3.