Added tests to gem