Clean up
Add assert check to verify returned values.
Add link to test class.