Add test initialization and cleanup