Expanded test suite

I added some tests.
Please let me know if there's stuff that you'd like to see change or if you have any wisdom to share :) I am happy to change anything as you see fit.