Fixing broken task
Minor style changes.