Added basic code for testing