Use appropriate WordPress functions