Adding adding support for custom post-types