Refactoring

Ein wenig Refactoring... sorry!