Wagtail 0.4 support

All done. Merge this when Wagtail 0.4 is released.