Change on the first class to be able to use Junit