Added a couple of unit tests
I'm tired