Bisschen refactoring