Cleaned up code according to the WordPress coding standard.