Adding first version of unit tests