Code aufgeräumt :)