fifth test working,
refactored code
fifth test working,
refactored code