Added some unittests

Added 2D unit tests