Added support for other post types.