Moved tests outside of main area. Added junit to gradle.