Unittest

Added the first unit test