Just added basic junit tests for reference