Made code follow Wordpress style guidelines