2nd try for assignment 4