Moved Unit tests to Test Folder

TODO: Include more test cases