Added test junit test :)