added one more test, made it passes refactored test