added week 2 code to refactor