Implement inital unit tests