Fixed JUnit tests and added TestAll class