Cleanup

Reformat code to match WordPress coding standards