Setting up unit testing using JUnit