Run tests after the main script