Add modified version of shunit2 script