flyttade lite kod.