Brought code in line with Wordpress guidelines