Added JUnit tests to solution.