docs: style and script changes for android u classes

Change-Id: I5866541bb9dd41329aa02328b805959c8f675f8c
diff --git a/tools/droiddoc/templates-sdk/assets/android-developer-docs.js b/tools/droiddoc/templates-sdk/assets/android-developer-docs.js
index 0cac2fc..c15584f 100644
--- a/tools/droiddoc/templates-sdk/assets/android-developer-docs.js
+++ b/tools/droiddoc/templates-sdk/assets/android-developer-docs.js
@@ -425,7 +425,7 @@
 
 $(window).unload(function(){
   var path = getBaseUri(location.pathname);
-  if (path.indexOf("/reference/") != -1) {
+  if ((path.indexOf("/reference/") != -1) || (path.indexOf("/training/") != -1)) {
     writeCookie("lastpage", path, "reference", null);
   } else if (path.indexOf("/guide/") != -1) {
     writeCookie("lastpage", path, "guide", null);