small refactor
Merge pull request #7 from BBCVisualJournalism/small_refactor

small refactor