Some cool feature (adding unit tests)

asdfasdf