Setup unit testing

A working one!