Minor update to tests, added docstrings

Requested by @sigmavirus24.