diff --git a/src/lambda.template.yaml b/src/lambda.template.yaml
index b725a85..e280280 100644
--- a/src/lambda.template.yaml
+++ b/src/lambda.template.yaml
@@ -12,6 +12,7 @@ Resources:
       Handler: 'index.@{HandlerName}'
       Role: !GetAtt LambdaExecRole.Arn
       Runtime: @{Runtime}
+      MemorySize: @{MemorySize}
       ReservedConcurrentExecutions: @{ReservedConcurrentExecutions}
       Timeout: @{Timeout}
       TracingConfig: {
diff --git a/src/lib/generate-cloudformation-template.ts b/src/lib/generate-cloudformation-template.ts
index ab76dea..9231463 100644
--- a/src/lib/generate-cloudformation-template.ts
+++ b/src/lib/generate-cloudformation-template.ts
@@ -1,7 +1,10 @@
 import * as fs from 'fs';
 import { UploadDeployOptions } from './../';
 
-export const generateCloudFormationTemplate = (options: UploadDeployOptions, lambdaExists: boolean) => {
+export const generateCloudFormationTemplate = (
+  options: UploadDeployOptions,
+  lambdaExists: boolean
+) => {
   const template = fs.readFileSync(
     `${__dirname}/../../src/lambda${
       lambdaExists ? '.version' : ''
@@ -15,63 +18,87 @@ export const generateCloudFormationTemplate = (options: UploadDeployOptions, lam
     .replace(/@{FunctionName}/g, options.functionName)
     .replace(/@{HandlerName}/g, options.handlerName)
     .replace(/@{Runtime}/g, options.settings.runtime)
-    .replace(/@{ReservedConcurrentExecutions}/g, (
-      !Number.isNaN(Number.parseInt(options.settings.reservedConcurrentExecutions as any)) 
-      ? options.settings.reservedConcurrentExecutions 
-      : '!Ref AWS::NoValue'
-    ).toString())
+    .replace(/@{MemorySize}/g, options.settings.memory)
+    .replace(
+      /@{ReservedConcurrentExecutions}/g,
+      (!Number.isNaN(
+        Number.parseInt(options.settings.reservedConcurrentExecutions as any)
+      )
+        ? options.settings.reservedConcurrentExecutions
+        : '!Ref AWS::NoValue'
+      ).toString()
+    )
     .replace(/@{Timeout}/g, options.settings.timeout.toString())
-    .replace(/@{VpcConfig}/g, options.settings.vpcConfig 
-      ? JSON.stringify(
-          Object.keys(options.settings.vpcConfig)
-            .reduce((acc, key) => {
-              acc[key.replace(/^\w/, _ => _.toUpperCase())] = (
-                (options.settings.vpcConfig as any)[key]
-              )
-              return acc;
-            }, {} as any)
-        )
-      : '!Ref AWS::NoValue'
+    .replace(
+      /@{VpcConfig}/g,
+      options.settings.vpcConfig
+        ? JSON.stringify(
+            Object.keys(options.settings.vpcConfig).reduce(
+              (acc, key) => {
+                acc[key.replace(/^\w/, _ => _.toUpperCase())] = (options
+                  .settings.vpcConfig as any)[key];
+                return acc;
+              },
+              {} as any
+            )
+          )
+        : '!Ref AWS::NoValue'
     )
     .replace(/@{Environment}/g, JSON.stringify(options.settings.environment))
     .replace(/@{Version}/g, options.version)
     .replace(/@{TracingMode}/g, options.settings.tracingConfig.mode)
-    .replace(/@{RoleStatement}/g, JSON.stringify([{
-      Effect: 'Allow',
-      Principal: {
-        Service: options.settings.servicesAllowed
-      },
-      Action: ['sts:AssumeRole']
-    }]))
-    .replace(/@{ManagedPolicies}/g, options.settings.managedPolicies
-      ? `ManagedPolicyArns: ${JSON.stringify(
-          options.settings.managedPolicies
-        )}`
-      : ''
+    .replace(
+      /@{RoleStatement}/g,
+      JSON.stringify([
+        {
+          Effect: 'Allow',
+          Principal: {
+            Service: options.settings.servicesAllowed
+          },
+          Action: ['sts:AssumeRole']
+        }
+      ])
+    )
+    .replace(
+      /@{ManagedPolicies}/g,
+      options.settings.managedPolicies
+        ? `ManagedPolicyArns: ${JSON.stringify(
+            options.settings.managedPolicies
+          )}`
+        : ''
     )
-    .replace(/@{PolicyStatement}/g, JSON.stringify(
-      options.settings.permissions.concat([{
-        effect: 'Allow',
-        action: [
-          'lambda:InvokeFunction'
-        ],
-        resource: [
-          `arn:aws:lambda:*:*:function:${options.functionName}`,
-          `arn:aws:lambda:*:*:function:${options.functionName}:*`
-        ]
-      },{
-        effect: 'Allow',
-        action: [
-          'logs:CreateLogGroup',
-          'logs:CreateLogStream',
-          'logs:PutLogEvents'
-        ],
-        resource: ['*']
-      }]).map((_:any) => Object.keys(_)
-        .reduce((acc, key) => {
-          acc[key.replace(/^\w/, _ => _.toUpperCase())] = _[key]
-          return acc;
-        }, {} as any)
+    .replace(
+      /@{PolicyStatement}/g,
+      JSON.stringify(
+        options.settings.permissions
+          .concat([
+            {
+              effect: 'Allow',
+              action: ['lambda:InvokeFunction'],
+              resource: [
+                `arn:aws:lambda:*:*:function:${options.functionName}`,
+                `arn:aws:lambda:*:*:function:${options.functionName}:*`
+              ]
+            },
+            {
+              effect: 'Allow',
+              action: [
+                'logs:CreateLogGroup',
+                'logs:CreateLogStream',
+                'logs:PutLogEvents'
+              ],
+              resource: ['*']
+            }
+          ])
+          .map((_: any) =>
+            Object.keys(_).reduce(
+              (acc, key) => {
+                acc[key.replace(/^\w/, _ => _.toUpperCase())] = _[key];
+                return acc;
+              },
+              {} as any
+            )
+          )
       )
-    ))
-};
\ No newline at end of file
+    );
+};