Switching to using jest directly

This is way easier than the plugin.