refactoring
challenge 4 done