tried out unit testing, not quite figure it out