Adding tests

You may want to add tests as you record.