UPDATE: Minor updates to readme
FEATURE: More unit tests