Added a couple more tests

I was writing an essay and got distracted by code.