Add support for JUnit (WIP)