Make class public.  Add junit