Added unit tests

Currently supported on python 2.7