Adding drush support