Adding in some jUnit