Added the first unit tests