doc update
Cleaned up unit tests