diff --git a/README.txt b/README.txt
index 55ade01..33e3375 100644
--- a/README.txt
+++ b/README.txt
@@ -9,7 +9,8 @@ CONTENTS OF THIS FILE
 ABOUT CUSTOM BODY CLASS
 ---------------------
 
-Custom body class module is used to add custom class to <body> tag specific to node.
+Custom body class module is used to add custom class to <body> tag specific to
+node.
 
 
 REQUIREMENTS
@@ -23,12 +24,12 @@ INSTALLATION
 
 CONFIGURATION
 -------------
-This module adds a fieldset named "Custom Body Class Settings" in node type form,
+This module adds a fieldset named "Custom Body Class Settings" in node type
+form,
 For eg: admin/structure/types/manage/article for article content type.
 
 There are two fields in a fieldset.
 1:  A textbox to add custom body class. Add multiple classes separated by space.
 2:  A checkbox,which when enabled add a node type as a class in <body> tag.
-    For eg: If a node type is article, then 'article' get added as a class in <body> tag of the page.
-
-
+    For eg: If a node type is article, then 'article' get added as a class in
+    <body> tag of the page.
diff --git a/custom_body_class.info.yml b/custom_body_class.info.yml
index 3048204..85b48ca 100644
--- a/custom_body_class.info.yml
+++ b/custom_body_class.info.yml
@@ -1,9 +1,4 @@
 type: module
 name : Custom body class
 description : Adds custom class/es
-# core : 8.x
-
-# Information added by Drupal.org packaging script on 2017-12-29
-version: '8.x-1.0'
-core: '8.x'
-project: 'custom_body_class'
+core : 8.x
diff --git a/custom_body_class.module b/custom_body_class.module
index 5675962..61d8daa 100644
--- a/custom_body_class.module
+++ b/custom_body_class.module
@@ -6,67 +6,66 @@
  *
  * This file adds a fieldset to the node form and adds class/es to the
  * node page in body tag.
-*/
+ */
 
 use Drupal\Core\Entity\EntityTypeInterface;
-use Drupal\Core\Field\FieldDefinition;
 use Drupal\Core\Field\BaseFieldDefinition;
 
 /**
  * Implements hook_form_FORM_ID_alter().
  */
-function custom_body_class_form_node_form_alter(&$form, \Drupal\Core\Form\FormStateInterface &$form_state, $form_id) {
-    $form['body_class_group'] = [
-        '#type' => 'details',
-        '#title' => t('Custom Body Class Settings'),
-        '#description' => t('Assign CSS class/es to the body.'),
-        '#group' => 'advanced',
-        '#access' => TRUE,
-        '#attributes' => ['class' => ['body-class-form']]
-    ];
+function custom_body_class_form_node_form_alter(&$form, &$form_state, $form_id) {
+  $form['body_class_group'] = [
+    '#type' => 'details',
+    '#title' => t('Custom Body Class Settings'),
+    '#description' => t('Assign CSS class/es to the body.'),
+    '#group' => 'advanced',
+    '#access' => TRUE,
+    '#attributes' => ['class' => ['body-class-form']],
+  ];
 
-    $form['body_class']['#group'] = 'body_class_group';
-    $form['specific_node_class']['#group'] = 'body_class_group';
+  $form['body_class']['#group'] = 'body_class_group';
+  $form['specific_node_class']['#group'] = 'body_class_group';
 }
 
 /**
  * Implements hook_entity_base_field_info().
  */
 function custom_body_class_entity_base_field_info(EntityTypeInterface $entity_type) {
-    if ($entity_type->id() === 'node') {
-        $fields['body_class'] = BaseFieldDefinition::create('string')
-        ->setLabel(t('Add CSS class(es)'))
-        ->setDisplayOptions('form', array(
-            'type' => 'string_textfield',
-            'weight' => 36,
-        ))
-        ->setDisplayConfigurable('form', TRUE);
-        
-        $fields['specific_node_class'] = BaseFieldDefinition::create('boolean')
-        ->setLabel(t('If checked, add name of node type as class to body tag.'))
-        ->setDisplayOptions('form', array(
-            'type' => 'boolean_checkbox',
-            'weight' => 37,
-        ))
-        ->setDisplayConfigurable('form', TRUE);
-        return $fields;
-    }
+  if ($entity_type->id() === 'node') {
+    $fields['body_class'] = BaseFieldDefinition::create('string')
+      ->setLabel(t('Add CSS class(es)'))
+      ->setDisplayOptions('form', [
+        'type' => 'string_textfield',
+        'weight' => 36,
+      ])
+      ->setDisplayConfigurable('form', TRUE);
+
+    $fields['specific_node_class'] = BaseFieldDefinition::create('boolean')
+      ->setLabel(t('If checked, add name of node type as class to body tag.'))
+      ->setDisplayOptions('form', [
+        'type' => 'boolean_checkbox',
+        'weight' => 37,
+      ])
+      ->setDisplayConfigurable('form', TRUE);
+    return $fields;
+  }
 }
 
 /**
  * Implements hook_preprocess_html().
  */
 function custom_body_class_preprocess_html(&$variables) {
-    $node = \Drupal::routeMatch()->getParameter('node');
-    
-    if($node){
-        $custom_body_class = $node->get('body_class')->getValue();
-        $specific_node_class = $node->get('specific_node_class')[0]->getValue();
-        if (!empty($custom_body_class) && isset($custom_body_class[0]) && isset($custom_body_class[0]['value'])) {
-          $variables['attributes']['class'][] = $custom_body_class[0]['value'];
-        }
-        if (!empty($specific_node_class) && ($specific_node_class['value'] == 1) ) {
-          $variables['attributes']['class'][] = $node->gettype();
-        }
+  $node = \Drupal::routeMatch()->getParameter('node');
+
+  if ($node) {
+    $custom_body_class = $node->get('body_class')->getValue();
+    $specific_node_class = $node->get('specific_node_class')[0]->getValue();
+    if (!empty($custom_body_class) && isset($custom_body_class[0]) && isset($custom_body_class[0]['value'])) {
+      $variables['attributes']['class'][] = $custom_body_class[0]['value'];
+    }
+    if (!empty($specific_node_class) && ($specific_node_class['value'] == 1)) {
+      $variables['attributes']['class'][] = $node->gettype();
     }
+  }
 }