Add a couple more tests. I'll move these to Selenium shortly.