Got the "types" jstree plugin working.

Switched to use it instead of doing it in php.