Adds tests to verify code