diff --git a/AMBuilder b/AMBuilder
index badf479..a86a866 100644
--- a/AMBuilder
+++ b/AMBuilder
@@ -6,13 +6,17 @@ projectName = 'socket'
 # smsdk_ext.cpp will be automatically added later
 sourceFiles = [
   'Extension.cpp',
+  'Callback.cpp',
+  'CallbackHandler.cpp',
+  'Socket.cpp',
+  'SocketHandler.cpp',
 ]
 
 ###############
 # Make sure to edit PackageScript, which copies your files to their appropriate locations
 # Simple extensions do not need to modify past this point.
 
-project = Extension.HL2Project(builder, projectName + '.ext')
+project = Extension.Library(builder, projectName + '.ext')
 
 if os.path.isfile(os.path.join(builder.currentSourcePath, 'sdk', 'smsdk_ext.cpp')):
   # Use the copy included in the project
@@ -22,10 +26,5 @@ else:
   project.sources += [os.path.join(Extension.sm_root, 'public', 'smsdk_ext.cpp')]
 
 project.sources += sourceFiles
-  
-for sdk_name in Extension.sdks:
-  sdk = Extension.sdks[sdk_name]
-  
-  binary = Extension.HL2Config(project, projectName + '.ext.' + sdk.ext, sdk)
 
-Extension.extensions = builder.Add(project)
+Extension.extensions = [builder.Add(project)]