Addressed comments for third assignment.