More tests

I added basic checking for it.