task 6 done after comments from Prof