Setting up a small framework for testing