Implemented first unit tests