Code cleanup to conform with WordPress coding style

Fixes #3