add code for local testing