Added first unit tests
Added first unit tests