Got unit tests actually working