Unittest

closes #2