Modified some class permisions
Added unit test
added missing files to unit test