Added JUNIT tests
changed int nr 3 to 4