Added some week 6 code