Adding first unit tests