Corrected code to fix Drupal Code Standards