Added first version of unit tests