assignment 6 almost finished