unit tests

noticed you didn't have the unit tests on your repo thought I would make
sure I got them into this one as well