Cleaned up code according to the WordPress coding standard.

Kunde inte låta bli att fixa till koden :) 
Hata inte mig.