Got started on some cool unit test stuff