Adds fifth test with refactor
Adds sixth test