Improve initialization script for test suite