test passed - added logic