Added first unit tests