Adding support for other post types