From 9cffbd171b3f1fdd61de251685984c885e5efadb Mon Sep 17 00:00:00 2001
From: Kiran Challa <kichalla@github.com>
Date: Wed, 25 Mar 2015 15:59:30 -0700
Subject: [PATCH] Fix Elm extensions

---
 .../ElmServiceCollectionExtensions.cs         | 23 +++++++++----------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/src/Microsoft.AspNet.Diagnostics.Elm/ElmServiceCollectionExtensions.cs b/src/Microsoft.AspNet.Diagnostics.Elm/ElmServiceCollectionExtensions.cs
index 7e4c9fb92c..34f0af3991 100644
--- a/src/Microsoft.AspNet.Diagnostics.Elm/ElmServiceCollectionExtensions.cs
+++ b/src/Microsoft.AspNet.Diagnostics.Elm/ElmServiceCollectionExtensions.cs
@@ -10,26 +10,25 @@ namespace Microsoft.Framework.DependencyInjection
     public static class ElmServiceCollectionExtensions
     {
         /// <summary>
-        /// Registers an <see cref="ElmStore"/> and configures <see cref="ElmOptions"/>.
+        /// Registers an <see cref="ElmStore"/> and configures default <see cref="ElmOptions"/>.
         /// </summary>
         public static IServiceCollection AddElm([NotNull] this IServiceCollection services)
         {
-            return services.AddElm(configureOptions: null);
+            services.AddOptions();
+            services.AddSingleton<ElmStore>();
+            return services;
         }
 
         /// <summary>
-        /// Registers an <see cref="ElmStore"/> and configures <see cref="ElmOptions"/>.
+        /// Configures a set of <see cref="ElmOptions"/> for the application.
         /// </summary>
-        public static IServiceCollection AddElm([NotNull] this IServiceCollection services, Action<ElmOptions> configureOptions)
+        /// <param name="services">The services available in the application.</param>
+        /// <param name="configureOptions">The <see cref="ElmOptions"/> which need to be configured.</param>
+        public static void ConfigureElm(
+            [NotNull] this IServiceCollection services, 
+            [NotNull] Action<ElmOptions> configureOptions)
         {
-            services.AddSingleton<ElmStore>(); // registering the service so it can be injected into constructors
-
-            if (configureOptions != null)
-            {
-                services.Configure(configureOptions);
-            }
-
-            return services;
+            services.Configure(configureOptions);
         }
     }
 }
\ No newline at end of file