Added tests for keys method