Added in support for custom posts.