Add a --test flag for easier testing