Fix object controller and add context category

NSManagedObjectContext+SSDataKitAdditions.h

@@ -0,0 +1,13 @@
+//
+//  NSManagedObjectContext+SSDataKitAdditions.h
+//  Disposamatic
+//
+//  Created by Sam Soffes on 1/25/12.
+//  Copyright (c) 2012 Synthetic. All rights reserved.
+//
+
+@interface NSManagedObjectContext (SSDataKitAdditions)
+
+- (NSManagedObjectContext *)newWithCurrentPersistentStoreCoordinator CF_RETURNS_RETAINED;
+
+@end

NSManagedObjectContext+SSDataKitAdditions.m

@@ -0,0 +1,19 @@
+//
+//  NSManagedObjectContext+SSDataKitAdditions.m
+//  Disposamatic
+//
+//  Created by Sam Soffes on 1/25/12.
+//  Copyright (c) 2012 Synthetic. All rights reserved.
+//
+
+#import "NSManagedObjectContext+SSDataKitAdditions.h"
+
+@implementation NSManagedObjectContext (SSDataKitAdditions)
+
+- (NSManagedObjectContext *)newWithCurrentPersistentStoreCoordinator {
+	NSManagedObjectContext *context = [[[self class] alloc] init];
+	context.persistentStoreCoordinator = self.persistentStoreCoordinator;
+	return context;
+}
+
+@end

SSDataKit.h

@@ -13,3 +13,5 @@
 #import "SSManagedObjectContext.h"
 #import "SSManagedObjectContextObserver.h"
 #import "SSManagedObjectController.h"
+
+#import "NSManagedObjectContext+SSDataKitAdditions.h"

SSManagedObjectController.m

@@ -19,6 +19,8 @@ @implementation SSManagedObjectController {
 
 - (id)init {
 	if ((self = [super init])) {
+		[self setup];
+
 		_observer = [[SSManagedObjectContextObserver alloc] init];
 		_observer.entity = self.entity;
 		_observer.observationBlock = ^(NSSet *insertedObjectIDs, NSSet *updatedObjectIDs) {