En lille refactor