Add the initial code for testing