Started unit tests
Added unit tests