Snyggade till (lite)

Fixade lite kod så det såg lite snyggare ut