Refactoring, image for day 5