Added wrapper function
Updated readme
Rearranged code - put testing code in main module