Adding mark test success as an optional feature