Updated tests.
Added doc strings.
CR feedback.
CR comments.