Add tests for new scripts