added functionality

and another test