Fixed tests

Fixed documentation