Added missing docs
More tests to get better coverage