added more code assgn 4