Adding test script (not finished)