diff --git a/like_and_dislike.module b/like_and_dislike.module
index 6823bf7..8f1ac68 100644
--- a/like_and_dislike.module
+++ b/like_and_dislike.module
@@ -1,9 +1,8 @@
 <?php
 
-/*
+/**
  * @file
- * This module provides 2 voting widgets: Like and Dislike. The behavior is
- * similar to Facebook and other social networks, but with negative rating also.
+ * This module provides 2 voting widgets: Like and Dislike.
  */
 
 use \Drupal\Core\Entity\EntityInterface;
@@ -91,11 +90,15 @@ function like_and_dislike_can_vote(AccountInterface $account, $vote_type_id, Ent
   || $account->hasPermission("add or remove $vote_type_id votes on {$entity->bundle()} of {$entity->getEntityTypeId()}");
 }
 
+/**
+ * Insert the like/dislike value if the user has select it.
+ */
 function like_and_dislike_vote_insert(EntityInterface $vote) {
   switch ($vote->bundle()) {
     case 'like':
       $vote_type_to_delete = 'dislike';
       break;
+
     case 'dislike':
       $vote_type_to_delete = 'like';
       break;
diff --git a/src/Controller/VoteController.php b/src/Controller/VoteController.php
index f15a43e..7680ccc 100644
--- a/src/Controller/VoteController.php
+++ b/src/Controller/VoteController.php
@@ -1,10 +1,5 @@
 <?php
 
-/**
- * @file
- * Contains \Drupal\like_and_dislike\Controller\VoteController.
- */
-
 namespace Drupal\like_and_dislike\Controller;
 
 use Drupal\Core\Controller\ControllerBase;
@@ -21,12 +16,22 @@ use Drupal\Core\Access\AccessResult;
  */
 class VoteController extends ControllerBase implements ContainerInjectionInterface {
 
-  protected $request;
-
   /**
-   * {@inheritdoc}
+   * Creates a vote for a given parameters.
+   *
+   * @param string $entity_type_id
+   *   The entity type ID to vote for.
+   * @param string $vote_type_id
+   *   The vote type (like or dislike).
+   * @param string $entity_id
+   *   The entity ID to vote for.
+   * @param \Symfony\Component\HttpFoundation\Request $request
+   *   The request.
+   *
+   * @return \Symfony\Component\HttpFoundation\JsonResponse
+   *   Returns JSON response.
    */
-  function vote($entity_type_id, $vote_type_id, $entity_id, Request $request) {
+  public function vote($entity_type_id, $vote_type_id, $entity_id, Request $request) {
     $entity = $this->entityTypeManager()->getStorage($entity_type_id)->load($entity_id);
 
     // Gets the number of likes and dislikes for the entity.
diff --git a/src/Tests/LikeAndDislikeTest.php b/src/Tests/LikeAndDislikeTest.php
index 9d45cfd..6418704 100644
--- a/src/Tests/LikeAndDislikeTest.php
+++ b/src/Tests/LikeAndDislikeTest.php
@@ -282,7 +282,7 @@ class LikeAndDislikeTest extends WebTestBase {
     $this->assertTrue(!isset($xpath['@class']));
     $this->assertEqual((string) $this->xpath('//*[@id="dislike-container-user-' . $user_id . '"]/a/@class')[0], 'disable-status');
 
-    // Assert that enabled_types is an empty Array
+    // Assert that enabled_types is an empty Array.
     $enabled_types = \Drupal::config('like_and_dislike.settings')->get('enabled_types');
     $this->assertEqual($enabled_types['user'], array());
   }