Added basic framework for testing