Added some tests! Yay!