Add unit tests

@park9140 Wanna pull in some unit tests?