Revert "Adding unit tests"

This reverts commit 3bd569147967b6322720ecedeb94d3b30b17235f.