Add boilerplate for testing