updated the tests and added rake task