refactor challenge week 6