Added some junit