Refactored something small